package com.qd.panda.service.msg;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.db.BaseService;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.RedisTemplateUtil;
import com.qd.common.panda.domain.entity.msg.UserMsgDTO;
import com.qd.common.panda.domain.entity.msg.UserMsgTypeDTO;
import com.qd.common.sys.util.SecurityUtils;
import com.qd.panda.mapper.msg.UserMsgMapper;
import com.qd.panda.model.UserMsg;
import com.qd.system.service.dict.MbUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.time.LocalDateTime;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

import static java.util.stream.Collectors.toList;

/**
 * 我的消息
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class UserMsgService extends BaseService<UserMsgMapper, UserMsg> {

    private static final String PREFIX_ID = "panda_user_msg_msgId_{0}";

    private final RedisTemplateUtil<UserMsg> redisTemplateUtil;

    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public UserMsgDTO save(UserMsgDTO dto) {
        final UserMsg model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public UserMsgDTO update(UserMsgDTO dto) {
        final UserMsg historyModel = Optional.ofNullable(getById(dto.getMsgId())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final UserMsg model = dtoToModel(dto);
        try {
            if (super.updateById(model)) {
                return modelToDto(model);
            }
            throw new ApiException("更新失败");
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
            model.freeData();
        }
    }

    /**
     * 删除，单个
     *
     * @param msgId 主键
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String msgId) {
        final UserMsg model = Optional.ofNullable(getById(msgId)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(msgId)) {
                throw new ApiException("删除失败");
            }
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(UserMsgDTO dto) {
        final QueryWrapper<UserMsg> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param msgId 主键
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String msgId) {
        boolean exists = false;
        if (StringUtils.hasLength(msgId)) {
            final UserMsg model = getById(msgId);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param msgId 主键
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public UserMsg getById(String msgId) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, msgId);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(msgId), UserMsg.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param msgId 主键
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public UserMsgDTO getByIdDto(String msgId) {
        return Optional.ofNullable(getById(msgId)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public UserMsg getOne(UserMsgDTO dto) {
        final QueryWrapper<UserMsg> queryWrapper = createQueryWrapper(dto);
        UserMsg model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public UserMsgDTO getOneDto(UserMsgDTO dto) {
        return Optional.ofNullable(getOne(dto)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据
     */
    @Transactional(readOnly = true)
    public List<UserMsg> getList(UserMsgDTO dto) {
        final QueryWrapper<UserMsg> queryWrapper = createQueryWrapper(dto);
        final List<UserMsg> list = super.list(queryWrapper);
        queryWrapper.clear();
        return list;
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据，转换成页面对象
     */
    @Transactional(readOnly = true)
    public List<UserMsgDTO> getListDto(UserMsgDTO dto) {
        return modelToDtoList(getList(dto));
    }

    /**
     * 分页数据
     *
     * @param page 分页对象
     * @param dto  查询参数
     * @return 分页信息
     */
    @Transactional(readOnly = true)
    public IPage<UserMsgDTO> page(Page<UserMsg> page, UserMsgDTO dto) {
        final QueryWrapper<UserMsg> queryWrapper = createQueryWrapper(dto, "ue.");
        final IPage<UserMsgDTO> iPage = baseMapper.page(page, queryWrapper);
        Optional.ofNullable(iPage.getRecords()).ifPresent(l -> l.forEach(this::buildDisplay));
        queryWrapper.clear();
        return iPage;
    }


    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(UserMsg model) {
        if (null == model) {
            return;
        }
        String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getMsgId());
        redisTemplateUtil.deleteByKey(key);
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public UserMsgDTO modelToDto(UserMsg model) {
        if (null == model) {
            return null;
        }
        final UserMsgDTO dto = new UserMsgDTO();
        dto.setMsgId(model.getMsgId());
        dto.setNote(model.getNote());
        dto.setAddTime(model.getAddTime());
        dto.setBid(model.getBid());
        dto.setCanRead(model.getCanRead());
        dto.setMsgType(model.getMsgType());
        dto.setUserId(model.getUserId());
        buildDisplay(dto);
        return dto;
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public UserMsg dtoToModel(UserMsgDTO dto) {
        if (null == dto) {
            return null;
        }
        final UserMsg model = new UserMsg();
        model.setMsgId(dto.getMsgId());
        model.setNote(dto.getNote());
        model.setAddTime(dto.getAddTime());
        model.setBid(dto.getBid());
        model.setCanRead(dto.getCanRead());
        model.setMsgType(dto.getMsgType());
        model.setUserId(dto.getUserId());
        return model;
    }

    /**
     * 码表转换
     *
     * @param dto 页面对象
     */
    public void buildDisplay(UserMsgDTO dto) {
        dto.setMsgTypeDisplay(MbUtil.idToDisplay(dto.getMsgType()));
    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<UserMsgDTO> modelToDtoList(List<UserMsg> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<UserMsg> dtoToModelList(List<UserMsgDTO> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<UserMsg> createQueryWrapper(UserMsgDTO dto) {
        return createQueryWrapper(dto, "");
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto        查询参数，不为空的属性会被自动加入
     * @param tableAlias 表别名
     */
    public QueryWrapper<UserMsg> createQueryWrapper(UserMsgDTO dto, String tableAlias) {
        final QueryWrapper<UserMsg> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(tableAlias)) {
                if (tableAlias.charAt(tableAlias.length() - 1) != '.') {
                    tableAlias += ".";
                }
            }
            if (null != dto.getCanRead()) {
                queryWrapper.eq(tableAlias + "can_read", dto.getCanRead());
            }
            if (StringUtils.hasLength(dto.getMsgType())) {
                queryWrapper.eq(tableAlias + "msg_type", dto.getMsgType());
            }
            if (StringUtils.hasLength(dto.getUserId())) {
                queryWrapper.eq(tableAlias + "user_id", dto.getUserId());
            }
        }
        queryWrapper.orderByDesc(tableAlias + "add_time");
        return queryWrapper;
    }

    /**
     * 批量检查主键在数据库中是否存在
     *
     * @param ids 主键集合
     */
    public List<UserMsg> checkIds(List<String> ids) {
        if (null == ids || ids.isEmpty()) {
            throw new ApiException("ids不能为空");
        }
        final List<UserMsg> list = super.listByIds(ids);
        if (list.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(ids.toArray()));
        }
        if (ids.size() == list.size()) {
            return list;
        }
        List<String> dbIds = list.stream().map(UserMsg::getMsgId).collect(toList());
        final List<String> notExistsList = ids.stream().filter(id -> !dbIds.contains(id)).collect(toList());
        dbIds.clear();
        if (!notExistsList.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(notExistsList.toArray()));
        }
        return list;
    }

    @Transactional(rollbackFor = Exception.class)
    public void addMsg(String msgType, String note) {
        UserMsgDTO userMsgDTO = new UserMsgDTO();
        userMsgDTO.setUserId(SecurityUtils.getLoginUser().getUserId());
        userMsgDTO.setMsgType(msgType);
        userMsgDTO.setNote(note);
        userMsgDTO.setAddTime(LocalDateTime.now());
        userMsgDTO.setCanRead(0);
        save(userMsgDTO);
        userMsgDTO.freeData();
    }

    /**
     * 标记消息状态为已读
     *
     * @param id 消息id，为空就表示标记这个人的所有消息为已读
     */
    @Transactional(rollbackFor = Exception.class)
    public void read(String id) {
        final String userId = SecurityUtils.getLoginUser().getUserId();
        if (null == id) {
            QueryWrapper<UserMsg> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("user_id", userId);
            final List<UserMsg> msgList = baseMapper.selectList(queryWrapper);
            if (!msgList.isEmpty()) {
                msgList.forEach(this::clearCache);
            }
            queryWrapper.eq("can_read", 0);
            getBaseMapper().update(new UserMsg().setUserId(userId).setCanRead(1), queryWrapper);
        } else {
            final UserMsgDTO userMsgDTO = getByIdDto(id);
            if (null != userMsgDTO) {
                if (!userId.equals(userMsgDTO.getUserId())) {
                    throw new ApiException("没有权限操作");
                }
                userMsgDTO.setCanRead(1);
                update(userMsgDTO);
                userMsgDTO.freeData();
            }
        }
    }

    @Transactional(readOnly = true)
    public List<UserMsgTypeDTO> msgTypeByUserId(String userId){
        return baseMapper.msgTypeByUserId(userId);
    }
}